#include "Http.hpp"

// 提供登录方法
void Login(HttpRquest &req, HttpResponse &resp)
{
    LOG(LogLevel::DEBUG) << req.Args() << ",我们成功进入到了处理数据的逻辑";
    std::string text = "hello: " + req.Args();

    // 这里其实应该做的是登录认证，要用到数据库，拿着用户名和密码在数据库中查询

    resp.SetCode(200);
    // 由于返回的是文本，就设置一下文本的Content-Type映射
    resp.SetHeader("Content-Type", "text/plain");
    // 再设置一下Content-Length
    resp.SetHeader("Content-Length", std::to_string(text.size()));
    // 最后填写一下正文
    resp.SetText(text);
}

// 提供注册方法
void Register(HttpRquest &req, HttpResponse &resp)
{
    LOG(LogLevel::DEBUG) << req.Args() << ",我们成功进入到了处理数据的逻辑";
    std::string text = "hello: " + req.Args();
    resp.SetCode(200);
    // 由于返回的是文本，就设置一下文本的Content-Type映射
    resp.SetHeader("Content-Type", "text/plain");
    // 再设置一下Content-Length
    resp.SetHeader("Content-Length", std::to_string(text.size()));
    // 最后填写一下正文
    resp.SetText(text);
}

// 提供检测vip用户的方法
void VipCheck(HttpRquest &req, HttpResponse &resp)
{
    LOG(LogLevel::DEBUG) << req.Args() << ",我们成功进入到了处理数据的逻辑";
    std::string text = "hello: " + req.Args();
    resp.SetCode(200);
    // 由于返回的是文本，就设置一下文本的Content-Type映射
    resp.SetHeader("Content-Type", "text/plain");
    // 再设置一下Content-Length
    resp.SetHeader("Content-Length", std::to_string(text.size()));
    // 最后填写一下正文
    resp.SetText(text);
}

// http port
int main(int argc, char *argv[])
{
    if (argc != 2)
    {
        std::cout << "Usage: " << argv[0] << " port" << std::endl;
        exit(USAGE_ERR);
    }

    uint16_t port = std::atoi(argv[1]);

    std::unique_ptr<Http> httpsvr = std::make_unique<Http>(port);
    // 新增登录服务
    httpsvr->RegisterService("/login", Login);
    // 新增各种服务
    httpsvr->RegisterService("/register", Register);
    httpsvr->RegisterService("/vip_check", VipCheck);

    httpsvr->Start();
    return 0;
}